from sklearn.tree import DecisionTreeClassifier, export_graphviz
from sklearn.datasets import load_iris

iris = load_iris()
X = iris.data
y = iris.target

tree = DecisionTreeClassifier(min_samples_leaf = 30, random_state = 10)
tree.fit(X,y)
export_graphviz(tree, out_file="iris.dot", class_names=iris.target_names,
                feature_names=iris.feature_names, filled=True)
